//给你一个整数数组 nums 和两个整数 indexDiff 和 valueDiff 。 
//
// 找出满足下述条件的下标对 (i, j)： 
//
// 
// i != j, 
// abs(i - j) <= indexDiff 
// abs(nums[i] - nums[j]) <= valueDiff 
// 
//
// 如果存在，返回 true ；否则，返回 false 。 
//
// 
//
// 示例 1： 
//
// 
//输入：nums = [1,2,3,1], indexDiff = 3, valueDiff = 0
//输出：true
//解释：可以找出 (i, j) = (0, 3) 。
//满足下述 3 个条件：
//i != j --> 0 != 3
//abs(i - j) <= indexDiff --> abs(0 - 3) <= 3
//abs(nums[i] - nums[j]) <= valueDiff --> abs(1 - 1) <= 0
// 
//
// 示例 2： 
//
// 
//输入：nums = [1,5,9,1,5,9], indexDiff = 2, valueDiff = 3
//输出：false
//解释：尝试所有可能的下标对 (i, j) ，均无法满足这 3 个条件，因此返回 false 。
// 
//
// 
//
// 提示： 
//
// 
// 2 <= nums.length <= 10⁵ 
// -10⁹ <= nums[i] <= 10⁹ 
// 1 <= indexDiff <= nums.length 
// 0 <= valueDiff <= 10⁹ 
// 
//
// Related Topics 数组 桶排序 有序集合 排序 滑动窗口 👍 749 👎 0


package LeetCode.editor.cn;


import java.util.HashMap;
import java.util.Map;
import java.util.TreeSet;

/**
 * @author ldltd
 * @date 2025-02-20 11:30:16
 * @description 220.存在重复元素 III
 
 */
 
public class ContainsDuplicateIii {
    public static void main(String[] args) {
    //测试代码
    ContainsDuplicateIii fun = new ContainsDuplicateIii();
    Solution solution= fun.new Solution();
    
    }

//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
    //维护一个k大小的滑动窗口
    //对于每个新加入的元素，比较窗口内是否有符合的
    //可以用有序集合来维护滑动窗口，这样比较时时间复杂度可以到onlogn
    public boolean containsNearbyAlmostDuplicate1(int[] nums, int k, int t) {
        int n = nums.length;
        TreeSet<Long> set = new TreeSet<Long>();
        for (int i = 0; i < n; i++) {
            //找到大于等于num[i]-t的最小值
            Long ceiling = set.ceiling((long) nums[i] - (long) t);
            if (ceiling != null && ceiling <= (long) nums[i] + (long) t) {
                return true;
            }
            set.add((long) nums[i]);
            if (i >= k) {
                set.remove((long) nums[i - k]);
            }
        }
        return false;
    }
    //滑动窗口+桶
    //维护一个窗口，窗口内的元素放入大小t+1的桶，对于每个新加入的元素，比较他应该放入的桶内是否有元素
    //如果没有，寻找是否有相邻桶中符合条件的元素，
    //每一个元素应该放入的桶位置，可以用 (t+1)*a+b表示，其中a是将要放入的桶
    //因为要放入的桶已经有元素了就可以返回了，所以每个桶可以用hash来维护

    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        int n = nums.length;
        Map<Long,Long> map=new HashMap<>();
        long w=(long) t+1;
        for (int i = 0; i < n; i++) {
            long ind=getInd(nums[i],w);
            if(map.containsKey(ind)) return true;
            if(map.containsKey(ind-1)&&Math.abs(nums[i]-map.get(ind-1))<w)return true;
            if(map.containsKey(ind+1)&&Math.abs(nums[i]-map.get(ind+1))<w) return true;
            map.put(ind,(long)nums[i]);
            if(i>=k){
                map.remove(getInd(nums[i-k],w));
            }
        }
        return false;
    }
    //桶从 1~t+1 t+2~2t+3
    //对于负数则上一个桶是 -t-1~0
    private long getInd(long num, long w) {
        if(num>=0){
            return num/w;
        }
        return (num+1)/w-1;
    }
}
//leetcode submit region end(Prohibit modification and deletion)

}
